import django_filters

from goods.models import FoodCategory, Foods


class FoodCategoryFilter(django_filters.rest_framework.FilterSet):

    shop_id = django_filters.CharFilter('shop_id')

    class Meta:
        model = FoodCategory
        fields = []


class FoodFilter(django_filters.rest_framework.FilterSet):

    food_id = django_filters.CharFilter('food_id')

    class Meta:
        model = Foods
        fields = []
